#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""PassServer项目清理脚本
用于删除与运行程序无关的测试代码和临时文件"""

import os
import shutil
import sys

# 设置工作目录
os.chdir(os.path.dirname(os.path.abspath(__file__)))

# 定义要删除的测试目录列表
test_dirs = [
    "update-service/src/test"  # Spring Boot项目的测试代码目录
]

# 定义要删除的可能的测试文件列表
test_files = [
    # 可以根据实际情况添加其他测试文件
]

def cleanup_files():
    """删除测试文件和目录"""
    print("===== 开始清理PassServer项目测试代码 =====")
    
    # 删除目录
    for dir_path in test_dirs:
        full_path = os.path.join(os.getcwd(), dir_path)
        if os.path.exists(full_path):
            try:
                shutil.rmtree(full_path)
                print(f"已删除测试目录: {dir_path}")
            except Exception as e:
                print(f"删除目录失败 {dir_path}: {e}")
        else:
            print(f"测试目录不存在: {dir_path}")
    
    # 删除文件
    for file_path in test_files:
        full_path = os.path.join(os.getcwd(), file_path)
        if os.path.exists(full_path):
            try:
                os.remove(full_path)
                print(f"已删除测试文件: {file_path}")
            except Exception as e:
                print(f"删除文件失败 {file_path}: {e}")
        else:
            print(f"测试文件不存在: {file_path}")
    
    print("\n===== PassServer项目测试代码清理完成 =====")
    print("\n保留的核心组件：")
    print("- eureka-server/ (服务注册中心)")
    print("- gateway/ (网关服务)")
    print("- update-service/ (更新服务，仅保留主代码)")
    print("- 配置文件和启动脚本")

if __name__ == "__main__":
    print("PassServer项目测试代码清理工具")
    print("此脚本将自动删除与运行程序无关的测试代码")
    
    # 直接执行清理
    cleanup_files()